import http.client
import json
import random

# 读取文件中的一行
file_path = r'c:\Users\Administrator\Desktop\工作项目\apollo_log_Chinese_byGPT\apollo_11_log.txt'  # 替换为你的文件路径

with open(file_path, 'r', encoding='utf-8') as file:
    lines = file.readlines()

# 从文件中随机选择一行
random_line = random.choice(lines).strip()  # 使用 .strip() 去掉行尾的换行符

# 调用翻译 API
conn = http.client.HTTPSConnection("api.v3.cm")
payload = json.dumps({
    "model": "gpt-4o",  # 确保使用有效的模型
    "messages": [
        {
            "role": "user",
            "content": f"翻译下述内容：{random_line}，选自阿波罗11号日志，只输出翻译内容"
        }
    ],
    "max_tokens": 1688,
    "temperature": 0.5,
    "stream": False
})

headers = {
    'Authorization': 'Bearer sk-hh02VJBqC5kv6ecFC40b3a282c5544E9908eEd30935e6d50',
    'Content-Type': 'application/json'
}

conn.request("POST", "/v1/chat/completions", payload, headers)
res = conn.getresponse()
data = res.read()
data.decode("utf-8")

# 解析为 JSON 对象
json_data = json.loads(data)

# 提取 "content" 内容（翻译后的文本）
content = json_data["choices"][0]["message"]["content"]

print(random_line,"\n",content)
